package com.saascloud.mybatis.core.handler;

import com.baomidou.mybatisplus.core.handlers.MetaObjectHandler;
import com.saascloud.core.SessionContext;
import com.saascloud.mybatis.core.entity.BaseFillEntity;
import org.apache.ibatis.reflection.MetaObject;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.Date;
import java.util.Objects;

/**
 * @Author: liangxu
 * @Contact: one_dream@aliyun.com
 * @Date: 2022/4/25 14:34
 * @Version: 1.0
 * @Description: 通用参数填充实现类
 * 如果没有显式的对通用参数进行赋值，这里会对通用参数进行填充、赋值,具体字段见
 * com.iss.mag.fr.db.com.iss.mag.fr.db.mp.core.entity.BaseFillEntity
 */
public class DefaultFieldHandler implements MetaObjectHandler {

    private SessionContext sessionContext;

    public DefaultFieldHandler(@Autowired(required = false) SessionContext sessionContext) {
        this.sessionContext = sessionContext;
    }

    @Override
    public void insertFill(MetaObject metaObject) {
        if (Objects.nonNull(metaObject) && metaObject.getOriginalObject() instanceof BaseFillEntity) {
            BaseFillEntity baseDO = (BaseFillEntity) metaObject.getOriginalObject();

            Date current = new Date();
            // 创建时间为空，则以当前时间为插入时间
            if (Objects.isNull(baseDO.getCreatedTime())) {
                baseDO.setCreatedTime(current);
            }
            // 更新时间为空，则以当前时间为更新时间
            if (Objects.isNull(baseDO.getUpdatedTime())) {
                baseDO.setUpdatedTime(current);
            }

            String userId = getUserId();
            // 当前登录用户不为空，创建人为空，则当前登录用户为创建人
            if (Objects.nonNull(userId) && Objects.isNull(baseDO.getCreatedBy())) {
                baseDO.setCreatedBy(userId);
            }
            // 当前登录用户不为空，更新人为空，则当前登录用户为更新人
            if (Objects.nonNull(userId) && Objects.isNull(baseDO.getUpdatedBy())) {
                baseDO.setUpdatedBy(userId);
            }
        }
    }

    @Override
    public void updateFill(MetaObject metaObject) {
        // 更新时间为空，则以当前时间为更新时间
        Object modifyTime = getFieldValByName("updatedTime", metaObject);
        if (Objects.isNull(modifyTime)) {
            setFieldValByName("updatedTime", new Date(), metaObject);
        }

        // 当前登录用户不为空，更新人为空，则当前登录用户为更新人
        Object modifier = getFieldValByName("updatedBy", metaObject);
        String userId = getUserId();
        if (Objects.nonNull(userId) && Objects.isNull(modifier)) {
            setFieldValByName("updatedBy", userId, metaObject);
        }
    }

    private String getUserId() {
        if (Objects.isNull(sessionContext)) {
            return null;
        }
        return sessionContext.getUser().getUserId();
    }
}
